package com.superiron.scaffold.config;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;

import java.util.List;

/**
 * @author ：yzg
 * @date ：Created in 2022/6/27 15:43
 * @description：
 * @modified By：
 * @version: $
 */
public class MySqlInjector extends DefaultSqlInjector {
	
	@Override
	public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
		
		List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);
		// 例: 不要指定了 update 填充的字段
		methodList.add(new InsertBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE));
		return methodList;
		
	}
}
